import os

seed = 163
new_path = r'E:\2021课程\机器视觉与图像测量\大作业\数据集\dataset_add2'


def __generate_name():
    global seed
    seed += 1
    return seed


# 改变图片名称
def change_name(path):
    file_name = 'img_add_' + str(__generate_name()).zfill(4) + '.jpg'
    global new_path
    oldname = path
    newname = new_path + os.sep + file_name
    if newname is not None:
        os.rename(oldname, newname)
        print(newname)


# 找到各文件夹下图片
def find_file(path):
    if os.path.exists(path):
        # 判断是文件时，更改名称
        if os.path.isfile(path):
            change_name(path)

        # 如果路径是文件，进入循环
        else:
            dirs = os.listdir(path)
            for dir in dirs:
                temp_path = path + os.sep + dir  # 这里不能直接用path = path +...，因为for每层递归结束时path保留，相当于path直接加上上一次的后缀
                find_file(temp_path)
                # print(path)
    else:
        return False


if __name__ == '__main__':
    path = r'E:\2021课程\机器视觉与图像测量\大作业\数据集\dataset_add'
    # path = r'E:\2021课程\机器视觉与图像测量\大作业\数据集\0003'
    print(find_file(path))

